package com.hyper_yang.algorithmRecord;

// 找出字符串中第一个匹配项的下标
public class LeetCode_28 {
    //    给你两个字符串 haystack 和 needle ，
//    请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标（下标从 0 开始）。
//    如果 needle 不是 haystack 的一部分，则返回  -1 。
/* 提示
    1 <= haystack.length, needle.length <= 10^4  (字符串都不为空)
    haystack 和 needle 仅由小写英文字符组成
 */
    // 方法一: 暴力法
    public static int strStr1(String haystack, String needle) {
        // 如果 needle 不是 haystack 的一部分
        if (haystack.length() < needle.length()) return -1;
        // 找第一个出现的字符，如果有肯定就在[0, haystack.length() - needle.length()]的范围
        for (int i = 0; i < haystack.length() - needle.length(); i++) {
            int j = 0;
            for (j = 0; j < needle.length(); j++) {
                if (haystack.charAt(i+j) != needle.charAt(j)) break; // 基本类型 ==
            }
            if (j == needle.length()) return i; // 匹配的长度 == 子串的长度（匹配上了），返回起始 index
        }
        return -1;
    }

    // 方法二: KMP算法 【很巧妙，多学习】
    public static int strStr(String haystack, String needle) {
        int n = haystack.length(), m = needle.length();
        if (n < m) return -1;

        int[] next = getNext(needle); // 依据 next数组，找最长公共前后缀
        int i = 0, j = 0;
        while (i < n) {
            if (haystack.charAt(i) == needle.charAt(j)) {
                i++;
                j++;
                if (j == m)
                    return i - m; // 找到匹配
            } else {
                if (j > 0)
                    j = next[j - 1]; // 根据 next数组回退
                else
                    i++; // j = 0且不匹配，主串指针前进
            }
        }
        return -1;
    }

    // 生成 next数组
    private static int[] getNext(String pattern) {
        int m = pattern.length();
        int[] next = new int[m];
        next[0] = 0;
        int j = 0;
        for (int i = 1; i < m; i++) { // 1个元素，没有公共前后缀
            while (j > 0 && pattern.charAt(i) != pattern.charAt(j)) {
                j = next[j - 1]; // 回退到前一个位置
            }
            if (pattern.charAt(i) == pattern.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }
    public static void main(String[] args) {
        System.out.println(strStr("sadbutsad", "sad")); // answer: 0
        System.out.println(strStr("leetcode", "leeto")); // answer: -1
    }
}
